NUM_CHANNELS = 3
IMG_SIZE = 256
